白話來說通過叠代地訓練決策樹每一棵樹都試圖糾正前一棵樹的預測錯誤從而逐步提升模型的性能
靈活性:GBDT 可以擬合各種類型的數據(數值型、類別型、文本等)。
高性能:在許多問題中,GBDT 提供了很高的預測準確度。
對異常值魯棒:由於采用了梯度下降的方式,GBDT 對異常值的影響相對較小。
from sklearn.ensemble import GradientBoostingClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# 載入鳶尾花數據集
iris = load_iris()
X, y = iris.data, iris.target
# 將數據集劃分為訓練集和測試集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 初始化GBDT分類器
gbdt_model = GradientBoostingClassifier(n_estimators=100, learning_rate=0.1, max_depth=3, random_state=42)
# 在訓練集上訓練模型
gbdt_model.fit(X_train, y_train)
# 在測試集上進行預測
y_pred = gbdt_model.predict(X_test)
# 計算準確度
accuracy = accuracy_score(y_test, y_pred)
print(f'準確度:{accuracy}')